#ifndef MODLE
#define MODLE

#include <torch/torch.h>

class Model{
public:
    Model();

    torch::Tensor input;
    torch::Tensor output;
    torch::Tensor loss;
    torch::Tensor predict;

    torch::Tensor weight;
    torch::Tensor bias;

    torch::Tensor dataSet;

    torch::Tensor forward(torch::Tensor input);
    int backward(torch::Tensor true_value);

};
#endif
